from Finance_fun.base.get_driver import GetDriver
from Finance_fun.page.page09_my_promotion import PageMyPromotion
import unittest


class MyPromotionTest(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = GetDriver.get_driver()
        cls.my_promotion = PageMyPromotion(cls.driver)
        cls.my_promotion.page_click_login_link()
        cls.my_promotion.page_login("13129628532","W123456")


    @classmethod
    def tearDownClass(cls):
        GetDriver.quit_driver()

    def test01_my_promotion(self):
        # 点击我的推广链接
        self.my_promotion.click_my_promotion_link()
        # 输入mq
        self.my_promotion.input_my_promotion_search("mq")
        # 点击搜索按钮
        self.my_promotion.click_my_promotion_search_btn()